
#include <iostream>
#include <fstream>
#include <string>
#include <string.h>
#include <vector>
#include <fstream>
#include <sys/stat.h>

#include "../source/server.hpp"
std::unordered_map<int, std::string> _status_msg = {
    {100, "Continue"},
    {101, "Switching Protocol"},
    {102, "Processing"},
    {103, "Early Hints"},
    {200, "OK"},
    {201, "Created"},
    {202, "Accepted"},
    {203, "Non-Authoritative Information"},
    {204, "No Content"},
    {205, "Reset Content"},
    {206, "Partial Content"},
    {207, "Multi-Status"},
    {208, "Already Reported"},
    {226, "IM Used"},
    {300, "Multiple Choice"},
    {301, "Moved Permanently"},
    {302, "Found"},
    {303, "See Other"},
    {304, "Not Modified"},
    {305, "Use Proxy"},
    {306, "unused"},
    {307, "Temporary Redirect"},
    {308, "Permanent Redirect"},
    {400, "Bad Request"},
    {401, "Unauthorized"},
    {402, "Payment Required"},
    {403, "Forbidden"},
    {404, "Not Found"},
    {405, "Method Not Allowed"},
    {406, "Not Acceptable"},
    {407, "Proxy Authentication Required"},
    {408, "Request Timeout"},
    {409, "Conflict"},
    {410, "Gone"},
    {411, "Length Required"},
    {412, "Precondition Failed"},
    {413, "Payload Too Large"},
    {414, "URI Too Long"},
    {415, "Unsupported Media Type"},
    {416, "Range Not Satisfiable"},
    {417, "Expectation Failed"},
    {418, "I'm a teapot"},
    {421, "Misdirected Request"},
    {422, "Unprocessable Entity"},
    {423, "Locked"},
    {424, "Failed Dependency"},
    {425, "Too Early"},
    {426, "Upgrade Required"},
    {428, "Precondition Required"},
    {429, "Too Many Requests"},
    {431, "Request Header Fields Too Large"},
    {451, "Unavailable For Legal Reasons"},
    {501, "Not Implemented"},
    {502, "Bad Gateway"},
    {503, "Service Unavailable"},
    {504, "Gateway Timeout"},
    {505, "HTTP Version Not Supported"},
    {506, "Variant Also Negotiates"},
    {507, "Insufficient Storage"},
    {508, "Loop Detected"},
    {510, "Not Extended"},
    {511, "Network Authentication Required"}};

std::unordered_map<std::string, std::string> _mine_msg = {
    {".aac", "audio/aac"},
    {".abw", "application/x-abiword"},
    {".arc", "application/x-freearc"},
    {".avi", "video/x-msvideo"},
    {".azw", "application/vnd.amazon.ebook"},
    {".bin", "application/octet-stream"},
    {".bmp", "image/bmp"},
    {".bz", "application/x-bzip"},
    {".bz2", "application/x-bzip2"},
    {".csh", "application/x-csh"},
    {".css", "text/css"},
    {".csv", "text/csv"},
    {".doc", "application/msword"},
    {".docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
    {".eot", "application/vnd.ms-fontobject"},
    {".epub", "application/epub+zip"},
    {".gif", "image/gif"},
    {".htm", "text/html"},
    {".html", "text/html"},
    {".ico", "image/vnd.microsoft.icon"},
    {".ics", "text/calendar"},
    {".jar", "application/java-archive"},
    {".jpeg", "image/jpeg"},
    {".jpg", "image/jpeg"},
    {".js", "text/javascript"},
    {".json", "application/json"},
    {".jsonld", "application/ld+json"},
    {".mid", "audio/midi"},
    {".midi", "audio/x-midi"},
    {".mjs", "text/javascript"},
    {".mp3", "audio/mpeg"},
    {".mpeg", "video/mpeg"},
    {".mpkg", "application/vnd.apple.installer+xml"},
    {".odp", "application/vnd.oasis.opendocument.presentation"},
    {".ods", "application/vnd.oasis.opendocument.spreadsheet"},
    {".odt", "application/vnd.oasis.opendocument.text"},
    {".oga", "audio/ogg"},
    {".ogv", "video/ogg"},
    {".ogx", "application/ogg"},
    {".otf", "font/otf"},
    {".png", "image/png"},
    {".pdf", "application/pdf"},
    {".ppt", "application/vnd.ms-powerpoint"},
    {".pptx", "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
    {".rar", "application/x-rar-compressed"},
    {".rtf", "application/rtf"},
    {".sh", "application/x-sh"},
    {".svg", "image/svg+xml"},
    {".swf", "application/x-shockwave-flash"},
    {".tar", "application/x-tar"},
    {".tif", "image/tiff"},
    {".tiff", "image/tiff"},
    {".ttf", "font/ttf"},
    {".txt", "text/plain"},
    {".vsd", "application/vnd.visio"},
    {".wav", "audio/wav"},
    {".weba", "audio/webm"},
    {".webm", "video/webm"},
    {".webp", "image/webp"},
    {".woff", "font/woff"},
    {".woff2", "font/woff2"},
    {".xhtml", "application/xhtml+xml"},
    {".xls", "application/vnd.ms-excel"},
    {".xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
    {".xml", "application/xml"},
    {".xul", "application/vnd.mozilla.xul+xml"},
    {".zip", "application/zip"},
    {".3gp", "video/3gpp"},
    {".3g2", "video/3gpp2"},
    {".7z", "application/x-7z-compressed"}};

static size_t Split(const std::string &src, const std::string &sep, std::vector<std::string> *arry)
{
    size_t offset = 0;
    while (offset < src.size())
    {
        size_t pos = src.find(sep, offset);
        if (pos == std::string::npos)
        {
            // abc.dfd.
            if (pos == src.size())
                break;
            arry->push_back(src.substr(offset));
            return arry->size();
        }
        if (pos == offset)
        {
            offset = pos + sep.size();
            continue;
            // abc,,,,,cd,
        }

        arry->push_back(src.substr(offset, pos - offset));
        offset = pos + sep.size();
    }
    return arry->size();
}
// 读取文件的所有内容，将读取的内容放到一个Buffer中
static bool ReadFile(const std::string &filename, std::string *buf)
{
    std::fstream ifs(filename, std::ios::binary | std::ios::in);
    if (ifs.is_open() == false)
    {
        printf("OPEN %s FILE FAILED!!", filename.c_str());
        return false;
    }
    size_t fsize = 0;
    ifs.seekg(0, ifs.end);       // 0是偏移量，将文件指针移到文件后面
    fsize = ifs.tellg();         // 获取文件指针与开头的总偏移量
    ifs.seekg(0, ifs.beg);       // 将文件指针移回开头
    buf->resize(fsize);          // 开辟文件大小的空间
    ifs.read(&(*buf)[0], fsize); //*buf是一个对象"hello" 而*buf[0]是字符‘h’,&(*buf)[0]就是首地址
    if (ifs.good() == false)
    {
        printf("READ %s FILE FAILED!!", filename.c_str());
        ifs.close();
        return false;
    }
    ifs.close();
    return true;
}

// 向文件写入数据
static bool WriteFile(const std::string &filename, const std::string &buf)
{
    std::ofstream ofs(filename, std::ios::binary | std::ios::out | std::ios::trunc); // 文件不存在，新增，存在，清空
    if (ofs.is_open() == false)
    {
        printf("OPEN %s FILE FAILED", filename.c_str());
    }
    ofs.write(buf.c_str(), buf.size());
    if (ofs.good() == false)
    {
        ERR_LOG("WRITE %s FILE FAILED !!", filename.c_str());
        ofs.close();
        return false;
    }
    ofs.close();
    return true;
}
static std::string UrlEncode(const std::string url, bool convert_space_to_plus)
{
    std::string res;
    for (auto &c : url)
    {
        if (c == '.' || c == '-' || c == '_' || c == '~' || isalnum(c))
        {
            res += c;
            continue;
        }
        if (c == ' ' && convert_space_to_plus)
        {
            res += '+';
            continue;
        }
        // 剩余的字符都是需要编码成为 %HH 格式
        char tmp[4] = {0};
        // %%转译为一个%, 0占位符, 2表示输出宽度为2,不足两位在那面补0
        snprintf(tmp, 4, "%%%02X", c);
        res += tmp;
    }
    return res;
}
static char HEXTOI(char c)
{
    if (c >= '0' && c <= '9')
    {
        return c - '0';
    }
    else if (c >= 'a' && c <= 'z')
    {
        return c - 'a' + 10;
    }
    else if (c >= 'A' && c <= 'Z')
    {
        return c - 'A' + 10;
    }
    return -1;
}
static std::string UrlDecode(const std::string url, bool convert_space_to_plus)
{
    std::string res;
    for (int i = 0; i < url.size(); i++)
    {
        if (url[i] == '+' && convert_space_to_plus)
        {
            res += ' ';
            continue;
        }
        if (url[i] == '%' && (i + 2) < url.size())
        {
            char v1 = HEXTOI(url[i + 1]);
            char v2 = HEXTOI(url[i + 2]);
            char v = v1 * 16 + v2;
            res += v;
            i += 2;
            continue;
        }
        res += url[i];
    }
    return res;
}
// 根据文件后缀名获取文件mine
static std::string ExtMine(const std::string &filename)
{

    // a.b.text 先获取文件扩展名
    size_t pos = filename.find_last_of('.');
    if (pos == std::string::npos)
    {
        return "application/octet-stream";
    }
    // 根据扩展名,获取mine
    std::string ext = filename.substr(pos);
    auto it = _mine_msg.find(ext);
    if (it == _mine_msg.end())
    {
        return "application/octet-stream";
    }

    return it->second;
}

void TestSplit()
{
    std::vector<std::string> result;
    // 测试用例1：正常分割（单个字符分隔符）
    std::string src1 = "hello,world,c++,Split";
    // 测试用例2：连续分隔符（中间有空字符串）
    std::string src2 = "a,,b,,,c";
    // 测试用例3：分隔符在字符串开头
    std::string src3 = ",prefix,test";
    // 测试用例4：分隔符在字符串结尾
    std::string src4 = "test,suffix,";
    // 测试用例5：使用多字符分隔符
    std::string src5 = "name=zhangsan;age=20;gender=male";
    std::string sep = ",";
    size_t count = Split(src5, sep, &result);
    std::cout << "分割结果（" << count << "个元素）：";
    for (auto &x : result)
    {
        std::cout << "[" << x << "]";
    }
    std::cout << "\n通过" << std::endl;
}
// 判断一个文件是否是一个目录
static bool IsDirectory(const std::string &filename)
{
    struct stat st;
    int ret = stat(filename.c_str(), &st);
    if (ret < 0)
    {
        return false;
    }
    return S_ISDIR(st.st_mode);
}
// 判断一个文件是否是一个普通文件
static bool IsRegular(const std::string &filename)
{
    struct stat st;
    int ret = stat(filename.c_str(), &st);
    if (ret < 0)
    {
        return false;
    }
    return S_ISREG(st.st_mode);
}
// http请求的资源路径有效性判断
// /index.html -- 前边的/ 叫做相对根目录， 映射的是某个服务器上的子目录
// 想表达的意思就是，客户端只能请求根目录中的资源，其他地方的资源不予理会
// /../login ,这个路径的..会让路径的查找跑到相对根目录之外，这是不合理的，不安全的
static bool ValidPath(const std::string &path)
{
    // 思想：按照/进行路径分割，根据有多少子目录，计算目录深度，有多少层，深度不能小于0
    std::vector<std::string> subdir;
    Split(path, "/", &subdir);
    int level = 0;
    for (auto &dir : subdir)
    {
        if (dir == "..")
        {
            level--; // 任意一层走出相对根目录，就认为有问题
            if (level < 0)
                return false;
            continue;
        }
        level++;
    }
    return true;
}
int main()
{
    std::cout << ValidPath("/html/../../index.html") <<std::endl;

    // std::cout << IsRegular("regex.cpp") << std::endl;
    // std::cout << IsRegular("textdir") << std::endl;
    // std::cout << IsDirectory("regex.cpp") << std::endl;
    // std::cout << IsDirectory("textdir") << std::endl;

    // std::cout << ExtMine("a.txt") << std::endl;

    // std::string str = "C  ";
    // std::string res = UrlEncode(str, false);
    // std::string tmp = UrlDecode(res, false);
    // std::cout<< "[" <<res <<"]"<< std::endl;
    // std::cout<< "["<< tmp <<"]"<< std::endl;

    // TestSplit();

    // std::string buf;
    // bool ret = ReadFile("./eventfd.c", &buf);
    // if (ret == false)
    // {
    //     return false;
    // }
    // ret = WriteFile("./ttttt.c", buf);
    // if(ret == false)
    // {
    //     return false;
    // }
    // std::cout << buf << std::endl;
    return 0;
}
